from sklearn.cluster import KMeans
import numpy as np

X = np.array([[1, 2], [1, 4], [1, 0],[10, 2], [10, 4], [10, 0]])
y=[1, 1, 1, 0, 0, 0]
kmeans = KMeans(n_clusters=2, random_state=0).fit(X,y)
# kmeans.labels_array([1, 1, 1, 0, 0, 0], dtype=int32)
# kmeans.predict([[0, 0], [12, 3]])array([1, 0], dtype=int32)
print(kmeans.predict([[0, 0], [12, 3]]))